﻿#ifndef KERNEL_CUH
#define KERNEL_CUH
extern "C" {
/**
 * @brief 初始化函数,用于预先分配cuda上的内存
 * @param w 矩阵的 width
 * @param h 矩阵的 height
 */
__declspec(dllexport) void initialize(int w, int h) ;
/**
 * @brief 释放已经分配的内存
 */
__declspec(dllexport) void myFree() ;

__declspec(dllexport) void foo(float *in, int w, int h, int sub, int purpose, float *out);
}
#endif